#ifndef Pump_H
#define Pump_H

#include "ItemBase.h"
#include "Pipe.h"

class Pump : public Pipe
{
	Q_OBJECT

public:
	Pump(const QString type, QPointF pos, ItemBase* linker);
	~Pump();

private:
	QString pumpType;
	QString pumpAttri;
	void paint(QPainter *painter, const QStyleOptionGraphicsItem * option, QWidget *widget);
	void initData();
	double angle;
	QPointF centerPos[3];

public:
	QMap<int, QString> getAttrs();
	QVector<bool> getDrawFlags();
	void refreshAngle();
};

#endif // Pump_H
